package com.test.utils;

import com.test.entity.Const;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Optional;
import java.util.concurrent.TimeUnit;

@Component
@Slf4j

public class FlowUtils {
    @Resource
    StringRedisTemplate template;
    /**
     * 检查流量是否超过限制
     *
     * 该方法用于检查给定key对应的流量是否超过了限制blockTime表示的时间间隔内只允许key对应的操作执行一次
     * 如果key已经存在，则认为流量超过限制，返回false；否则，设置key，并认为流量未超过限制，返回true
     *
     * @param key 用于标识操作的唯一键，通常使用IP地址或其他标识符
     * @param blockTime 在该时间间隔内，只允许key对应的操作执行一次，单位为秒
     * @return 如果流量超过限制，则返回false；否则返回true
     */
    public boolean checkFlow(String key,int blockTime){
        // 检查key是否存在，如果存在则认为流量超过限制，返回false
        if(template.hasKey(key)){
            return false;
        }
        else {
            // 如果key不存在，则设置key，并设置过期时间为blockTime秒，认为流量未超过限制，返回true
            template.opsForValue().set(key,"",blockTime, TimeUnit.SECONDS);
            return true;
        }
    }
    public boolean limitPeriodCounterCheck(String counterKey,int frequency,int period){
        // 检查是否存在该IP的请求计数
        if(Boolean.TRUE.equals(template.hasKey(counterKey))){
            // 增加请求计数
            Long increment = Optional.ofNullable(template.opsForValue().increment(counterKey)).orElse((0L));
            // 检查增加后的请求计数是否超过了限制
            if(increment > frequency){
                log.info(counterKey + "访问过于频繁");
                log.info("计数器：" + template.opsForValue().increment(counterKey));
                return false;
            }
        }
        else{
            // 如果不存在请求计数，设置初始计数，并设置过期时间
            template.opsForValue().set(counterKey, "", period, TimeUnit.SECONDS);

        }
        // 如果请求计数没有超过限制，返回true
        return true;
    }

}
